<?php

namespace VendorDuplicator\JmesPath;

/**
 * Provides CLI debugging information for the AST and Compiler runtimes.
 */
class DebugRuntime
{
    private $runtime;
    private $out;
    private $lexer;
    private $parser;
    public function __construct(callable $runtime, $output = null)
    {
        $this->runtime = $runtime;
        $this->out = $output ?: \STDOUT;
        $this->lexer = new Lexer();
        $this->parser = new Parser($this->lexer);
    }
    public function __invoke($expression, $data)
    {
        if ($this->runtime instanceof CompilerRuntime) {
            return $this->debugCompiled($expression, $data);
        }
        return $this->debugInterpreted($expression, $data);
    }
    private function debugInterpreted($expression, $data)
    {
        return $this->debugCallback(function () use($expression, $data) {
            $runtime = $this->runtime;
            return $runtime($expression, $data);
        }, $expression, $data);
    }
    private function debugCompiled($expression, $data)
    {
        $result = $this->debugCallback(function () use($expression, $data) {
            $runtime = $this->runtime;
            return $runtime($expression, $data);
        }, $expression, $data);
        $this->dumpCompiledCode($expression);
        return $result;
    }
    private function dumpTokens($expression)
    {
        $lexer = new Lexer();
        \fwrite($this->out, "Tokens\n======\n\n");
        $tokens = $lexer->tokenize($expression);
        foreach ($tokens as $t) {
            \fprintf($this->out, "%3d  %-13s  %s\n", $t['pos'], $t['type'], \json_encode($t['value']));
        }
        \fwrite($this->out, "\n");
    }
    private function dumpAst($expression)
    {
        $parser = new Parser();
        $ast = $parser->parse($expression);
        \fwrite($this->out, "AST\n========\n\n");
        \fwrite($this->out, \json_encode($ast, \JSON_PRETTY_PRINT) . "\n");
    }
    private function dumpCompiledCode($expression)
    {
        \fwrite($this->out, "Code\n========\n\n");
        $dir = \sys_get_temp_dir();
        $hash = \md5($expression);
        $functionName = "jmespath_{$hash}";
        $filename = "{$dir}/{$functionName}.php";
        \fwrite($this->out, "File: {$filename}\n\n");
        \fprintf($this->out, \file_get_contents($filename));
    }
    private function debugCallback(callable $debugFn, $expression, $data)
    {
        \fprintf($this->out, "Expression\n==========\n\n%s\n\n", $expression);
        $this->dumpTokens($expression);
        $this->dumpAst($expression);
        \fprintf($this->out, "\nData\n====\n\n%s\n\n", \json_encode($data, \JSON_PRETTY_PRINT));
        $startTime = \microtime(\true);
        $result = $debugFn();
        $total = \microtime(\true) - $startTime;
        \fprintf($this->out, "\nResult\n======\n\n%s\n\n", \json_encode($result, \JSON_PRETTY_PRINT));
        \fwrite($this->out, "Time\n====\n\n");
        \fprintf($this->out, "Total time:     %f ms\n\n", $total);
        return $result;
    }
}
